# -*- coding: utf-8 -*-

import wget
from otherpkg.utils import unzip_file

if __name__ == "__main__":
    train_filename = wget.download(
        "https://tesla-ap-shanghai-1256322946.cos.ap-shanghai.myqcloud.com/cephfs/tesla_common/deeplearning/dataset/algo_contest/train_preliminary.zip")
    test_filename = wget.download(
        "https://tesla-ap-shanghai-1256322946.cos.ap-shanghai.myqcloud.com/cephfs/tesla_common/deeplearning/dataset/algo_contest/test.zip")
    print(train_filename)
    print(test_filename)
    # 解压文件
    output_dir = './input'
    unzip_file(train_filename, output_dir)
    unzip_file(test_filename, output_dir)
